package com.vr.cloudpicture.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;


/**
 * 线程池配置
 *
 * @author hzh
 * @date 2025/05/23
 */
@Configuration
@EnableAsync
public class ThreadPoolConfig {
    @Bean
    public ThreadPoolTaskExecutor customExecutor() {
        // 核心线程数设置为CPU核心数的2倍，以充分利用计算资源
        int corPoolSize = Runtime.getRuntime().availableProcessors() * 2;
        // 创建线程池执行器实例
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        // 设置核心线程数
        executor.setCorePoolSize(corPoolSize);
        // 设置最大线程数为核心线程数的2倍，允许在负载高时扩展
        executor.setMaxPoolSize(corPoolSize * 2);
        // 设置队列容量为1000，用于存放等待执行的任务
        executor.setQueueCapacity(1000);
        // 设置线程名称前缀，便于日志跟踪和问题排查
        executor.setThreadNamePrefix("custom-executor-");
        // 设置任务完成后关闭线程池时等待所有任务完成
        executor.setWaitForTasksToCompleteOnShutdown(true);
        // 设置线程池关闭时等待任务完成的最大时间（秒）
        executor.setAwaitTerminationSeconds(60);
        return executor;
    }
}
